from intel_extension_for_pytorch.utils.utils import has_cpu

if has_cpu():
    from ...cpu.nn.frozen_batch_norm import FrozenBatchNorm2d
    from ...cpu.nn import _roi_align
    from .merged_embeddingbag import MergedEmbeddingBagWithSGD
    from .merged_embeddingbag import MergedEmbeddingBag
    from .merged_embeddingbag import MergedEmbeddingBagWithCat
    from .merged_embeddingbag import MergedEmbeddingBagWithAdaGrad
    from .merged_embeddingbag import DistMergeEmbeddingBagWithAdaGrad
    from ...cpu.nn.linear_fuse_eltwise import IPEXLinearEltwise

from .weight_only_quantization import WeightOnlyQuantizedLinear
